from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-pose.pt')  # load a pretrained model (recommended for training)
# model = YOLO('yolov8x-pose.pt')  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data='./datasets/pose.yaml', epochs=100, imgsz=640)
# results = model.train(data='./datasets/pose_flip_idx.yaml', epochs=100, imgsz=640)